from core.layer import Layer
from utils.dot_graph import plot_dot_graph

# model继承layer
# 用式子抽象描述潜藏着复杂模式或规则的现象
class Model(Layer):
    # 可视化操作，生成计算图
    def plot(self, *inputs, to_file='model.png'):
        y = self.forward(*inputs)
        return plot_dot_graph(y, verbose=True, to_file=to_file)
